# -*- coding:utf-8 -*-
"""
作者：520
日期：2023年12月07日
"""
import numpy as np
import tensorflow as tf

opt = tf.keras.optimizers.SGD(lr=0.1, momentum=0.9)
var = tf.Variable(1.0)
val0 = var.value()
loss = lambda: (var ** 2) / 2.0
opt.minimize(loss, [var])
val1 = var.value()
opt.minimize(loss, [var])
val2 = var.value()
print("第一次步长={}".format((val0 - val1)))
print("第一次步长={}".format((val1 - val2)))


